package me.mingshan.leetcode;

import java.util.Arrays;
import java.util.Stack;

/**
 * 删除重复丶数组
 * <p>
 * 给你一个有序数组 nums ，请你 原地 删除重复出现的元素，使每个元素 最多出现两次 ，返回删除后数组的新长度。
 * <p>
 * 不要使用额外的数组空间，你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
 * <p>
 * 示例 1：
 * <p>
 * 输入：nums = [1,1,1,2,2,3]
 * 输出：5, nums = [1,1,2,2,3]
 * 解释：函数应返回新长度 length = 5, 并且原数组的前五个元素被修改为 1, 1, 2, 2, 3 。 不需要考虑数组中超出新长度后面的元素。
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array-ii
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 * <p>
 * <p>
 * 未实现
 */
public class L_80_RemoveDuplicates {

    public static void main(String[] args) {
//    int[] nums = {1, 1, 1, 2, 2, 3};
//    System.out.println(removeDuplicates(nums));

        int[] nums2 = {0, 0, 1, 1, 1, 1, 2, 3, 3};
        //System.out.println(removeDuplicates2(nums2));

        int[] nums3 = {1, 1, 1, 2, 2, 3};
        System.out.println(removeDuplicates2(nums3));
    }

    /**
     * 思路：
     * 由于同一个数，至少保留两个，所以
     * 1. 创建一个指针，指向当前位置
     * 2. 创建一个指针，指向下一个待处理的位置
     * 3. 需要增加个计数器，需要知道相同数字重复的个数，需要保留两个内的重复数字
     *
     * @param nums
     * @return
     */
    public static int removeDuplicates(int[] nums) {
        if (nums == null || nums.length == 0) {
            return 0;
        }

        // 指向新数组最后的位置
        int left = 0;
        // 指向下一个待处理的位置
        int right = 1;

        // 代表right扫描到相同数字的个数
        int count = 1;

        for (right = 1; right < nums.length; right++) {
            // 如果下一个与上一个相等
            if (nums[right] == nums[left]) {
                // 先判断计数器，如果小于等于2
                // 小于2，继续往前走
                if (count < 2) {
                    left++;
                    if (left != right) {
                        nums[left] = nums[right];
                    }
                } else {
                    // 如果等于2，说明left不能再往前走，要找下一个不一样的
                    // NOOP
                }
                count++;
            } else {
                // 如果下一个与上一个不相等
                left++;
                nums[left] = nums[right];
                count = 1;
            }
        }

        System.out.println(Arrays.toString(nums));

        return left + 1;
    }

    /**
     * 使用栈
     *
     * {0, 0, 1, 1, 1, 1, 2, 3, 3
     *
     * 每次都与栈顶的下一个元素比较，如果不相等，则入栈
     *
     * @param nums
     * @return
     */
    public static int removeDuplicates2(int[] nums) {
        if (nums == null || nums.length == 0) {
            return 0;
        }

        if (nums.length <= 2) {
            return nums.length;
        }

        // 栈顶
        int left = 1;

        for (int i = 2; i < nums.length; i++) {
            if (nums[i] != nums[left - 1]) {
                left++;
                nums[left] = nums[i];
            }
        }

        System.out.println(Arrays.toString(nums));

        return left + 1;
    }

}
